package com.siecom.algorithm.aes;

import android.os.Build;


import com.siecom.tool.ByteTool;

import java.nio.charset.Charset;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

public class AESUtils {

   private static final String TAG = AESUtils.class.getSimpleName();

   /**
    * 采用AES加密算法
    */
   private static final String KEY_ALGORITHM = "AES";

   /**
    * 字符编码(用哪个都可以，要注意new String()默认使用UTF-8编码 getBytes()默认使用ISO8859-1编码)
    */
   private static final Charset CHARSET_UTF8 = Charset.forName("UTF-8");

   /**
    * 加解密算法/工作模式/填充方式
    */
   private static final String CIPHER_ALGORITHM = "AES/CBC/PKCS5Padding";

   /**
    * AES 加密
    *
    * @param secretKey 加密密码，长度：16 或 32 个字符
    * @param data      待加密内容
    * @return 返回Base64转码后的加密数据
    */
   public static String encrypt(byte[] secretKey, byte[] iv, String data) {
      try {
         // 创建AES秘钥
         SecretKeySpec secretKeySpec = new SecretKeySpec(secretKey, KEY_ALGORITHM);
         // 创建密码器
         Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
         // 初始化加密器
         cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, new IvParameterSpec(iv));
         byte[] encryptByte = cipher.doFinal(data.getBytes(CHARSET_UTF8));
         // 将加密以后的数据进行 Base64 编码
         //return ByteTool.byte2hex(encryptByte);
         return base64Encode(encryptByte);
      } catch (Exception e) {
         e.printStackTrace();
      }
      return null;
   }

   /**
    * AES 解密
    *
    * @param secretKey  解密的密钥，长度：16 或 32 个字符
    * @param base64Data 加密的密文 Base64 字符串
    */
   public static String decrypt(byte[] secretKey, byte[] iv, String base64Data) {
      try {
         byte[] data = base64Decode(base64Data);
         //byte[] data = ByteTool.hexStr2Bytes(base64Data);
         // 创建AES秘钥
         SecretKeySpec secretKeySpec = new SecretKeySpec(secretKey, KEY_ALGORITHM);
         // 创建密码器
         Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
         // 初始化解密器
         cipher.init(Cipher.DECRYPT_MODE, secretKeySpec, new IvParameterSpec(iv));
         // 执行解密操作
         byte[] result = cipher.doFinal(data);
         return new String(result, CHARSET_UTF8);
      } catch (Exception e) {
         e.printStackTrace();
      }
      return null;
   }

   /**
    * 将 字节数组 转换成 Base64 编码
    * 用Base64.DEFAULT模式会导致加密的text下面多一行（在应用中显示是这样）
    */
   public static String base64Encode(byte[] data) {
      if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
         return new String(java.util.Base64.getEncoder().encode(data), CHARSET_UTF8);
      }else {
         //return new String(android.util.Base64.encode(data, android.util.Base64.NO_WRAP), CHARSET_UTF8);
         return ByteTool.byte2hex(data);
      }
   }

   /**
    * 将 Base64 字符串 解码成 字节数组
    */
   public static byte[] base64Decode(String data) {
      if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
         return java.util.Base64.getDecoder().decode(data.getBytes(CHARSET_UTF8));
      }else{
         //return android.util.Base64.decode(data.getBytes(CHARSET_UTF8), android.util.Base64.NO_WRAP);
         return ByteTool.hexStr2Bytes(data);
      }
   }

}
